package com.chaosj.algorithm

/**
 * @Description:
 * @author: caoyangjie
 * @date: 2019年11月2019/11/14日 16:27
 * @Copyright: © 赢时胜
 */
class TwoSum {
    /**
     * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
     *
     * 你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
     *
     * 示例:
     *
     * 给定 nums = [2, 7, 11, 15], target = 9
     *
     * 因为 nums[0] + nums[1] = 2 + 7 = 9
     * 所以返回 [0, 1]
     */
    static void main(String[] args){
        int[] nums = Arrays.asList(2,7,11,15).toArray(new int[4])
        int target = 9
        System.out.println(twoSum(nums,target))
    }

    static int[] twoSum(int[] nums, int target) {
        //第一种解决方案, 暴力法
        System.out.println(solution1(nums,target))
        //第二种解决方案, 一遍哈希法
        solution2(nums,target)
    }

    static int[] solution1(int[] nums,int target){
        int[] solution = new int[2]
        for (int i = 0; i < nums.length; i++) {
            for (int j = 0; j < nums.length; j++) {
                if( nums[i] == target-nums[j] ){
                    solution[0] = i
                    solution[1] = j
                    return solution
                }
            }
        }
        throw new IllegalArgumentException("No two sum solution");
    }

    static int[] solution2(int[] nums,int target){
        int[] solution = new int[2]
        Map<Integer,Integer> data = new HashMap<>(nums.size())
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i]
            if( data.containsKey(complement) ){
                solution[0] = data.get(complement)
                solution[1] = i
                return solution
            }
            data.put(nums[i],i)
        }
    }
}
